package model

import (
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"time"
)

var DB *gorm.DB

func Init() {
	DB, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{
		// 开启 WAL 模式
		//DSN: "mode=wal",
		// 增加最大连接数为 100
		//MaxOpenConns: 100,
	})
	if err != nil {
		panic(err)
	}
	// 设置数据库连接池参数
	sqlDB, err := DB.DB()
	if err != nil {
		panic(err)
	}
	sqlDB.SetMaxIdleConns(10)
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetConnMaxLifetime(time.Hour)

	//err = Db.Migrator().CreateTable(&User{})
	err = DB.AutoMigrate(&User{}, &Shape{})
	if err != nil {
		panic(err)
	}
}
